import tensorflow as tf

gpus = tf.config.list_physical_devices(device_type='GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)


class StockPredictModel0(tf.keras.Model):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.dense_1 = tf.keras.layers.Dense(
            1,
            use_bias=False,
            kernel_initializer=tf.random_normal_initializer(stddev=0.01)
        )

    def build(self, batch_size: int):
        super().build(input_shape=[
            (batch_size, None, 1),
        ])

    def call(self, inputs, past=None, **kwargs):
        return self.dense_1(inputs[0]) * 0
